﻿using System;
using System.Collections.Generic;
using System.ComponentModel.Composition.Hosting;
using System.ComponentModel.Composition.Primitives;
using System.Linq;
using System.Reflection;
using Caliburn.Micro;
using NathansMapInfoProjects.PluginManager.Utils;
using NathansMapInfoProjects.PluginManager.ViewModels;
using System.ComponentModel.Composition;

namespace NathansMapInfoProjects.PluginManager
{
    public class ManagerBootstrapper : Bootstrapper<IShell>
    {
        private CompositionContainer container;

        static ManagerBootstrapper()
        {
#if DEBUG
            LogManager.GetLog = type => new DebugLogger(type);
#endif
        }

        protected override void Configure()
        {
            container = CompositionHost.Initialize(
                new AggregateCatalog(
                        AssemblySource.Instance.Select(x => new AssemblyCatalog(x)).OfType<ComposablePartCatalog>()
                    )
                );

            var batch = new CompositionBatch();
            // Add the window manager and event aggregator to the IoC so that things can import them.
            batch.AddExportedValue<IWindowManager>(new WindowManager());
            batch.AddExportedValue<IEventAggregator>(new EventAggregator());
            batch.AddExportedValue(container);

            container.Compose(batch);
        }

        protected override object GetInstance(Type serviceType, string key)
        {
            string contract = string.IsNullOrEmpty(key) ? AttributedModelServices.GetContractName(serviceType) : key;
            var exports = container.GetExportedValues<object>(contract);

            if (exports.Count() > 0)
                return exports.First();

            throw new Exception(string.Format("Could not locate any instances of contract {0}.", contract));
        }

        protected override IEnumerable<object> GetAllInstances(Type serviceType)
        {
            return container.GetExportedValues<object>(AttributedModelServices.GetContractName(serviceType));
        }

        protected override void BuildUp(object instance)
        {
            container.SatisfyImportsOnce(instance);
        }

        protected override IEnumerable<Assembly> SelectAssemblies()
        {
            return new[] {
                            Assembly.GetAssembly(typeof(ManagerBootstrapper)),
                            Assembly.GetEntryAssembly()
                         };
        }
    }
}